import logging
from .ntu_generator import NTU_Generator
__generator = {
    'ntu': NTU_Generator,
}

def create(args):
    dataset = args.dataset.split('-')[0]
    if dataset not in __generator.keys():
        logging.info('')
        logging.error('Error: Do NOT exist this dataset: {}!'.format(dataset))
        raise ValueError()
    return __generator[dataset](args)